Armstrong number in python